# Copyright (c) 2020-2021, NVIDIA CORPORATION. All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
#  * Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
#  * Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
#  * Neither the name of NVIDIA CORPORATION nor the names of its
#    contributors may be used to endorse or promote products derived
#    from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ``AS IS'' AND ANY
# EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
# PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
# PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
# OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

cmake_minimum_required(VERSION 3.17)

project(tritonpreprocessbackend LANGUAGES C CXX)

set(CMAKE_C_COMPILER "/usr/local/corex/bin/clang")
set(CMAKE_CXX_COMPILER "/usr/local/corex/bin/clang++")
#SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -w")

#
# Options
#
set(TRITON_BACKEND_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/backend repo.")
set(TRITON_COMMON_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/common repo")
set(TRITON_CORE_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/core repo")

if(NOT CMAKE_BUILD_TYPE)
  set(CMAKE_BUILD_TYPE Release)
endif()

set(TRITON_PREPROCESS_BACKEND_LIBNAME triton_preprocess)

#
# Dependencies
#
include(FetchContent)

message(WARNING "fetching tis-backend:https://gitee.com/121786404/backend.git")
FetchContent_Declare(
  repo-backend
  GIT_REPOSITORY https://gitee.com/121786404/backend.git
  GIT_TAG ${TRITON_BACKEND_REPO_TAG}
)

message(WARNING "fetching repo-backend:https://gitee.com/121786404/common.git")
FetchContent_Declare(
  repo-common
  GIT_REPOSITORY https://gitee.com/121786404/common.git
  GIT_TAG ${TRITON_COMMON_REPO_TAG}
  GIT_SHALLOW ON
)

message(WARNING "fetching tis-core:https://gitee.com/121786404/core.git")
FetchContent_Declare(
  repo-core
  GIT_REPOSITORY https://gitee.com/121786404/core.git
  GIT_TAG ${TRITON_CORE_REPO_TAG}
  GIT_SHALLOW ON
)
FetchContent_MakeAvailable(repo-backend repo-common repo-core)

#
# Backend library containing useful source and utilities
#
add_library(
  triton-preprocess-backend SHARED
  src/preprocess.cc
  src/preprocess.h
)

add_library(
  TritonPreProcessBackend::triton-preprocess-backend ALIAS triton-preprocess-backend
)

target_compile_features(triton-preprocess-backend PRIVATE cxx_std_14)
# target_compile_options(
#  triton-preprocess-backend
#  PRIVATE
#  -w
# )


set_target_properties(
  triton-preprocess-backend PROPERTIES
  POSITION_INDEPENDENT_CODE ON
  OUTPUT_NAME ${TRITON_PREPROCESS_BACKEND_LIBNAME}
)


# which will cause core dump
# add_definitions(-D_GLIBCXX_USE_CXX11_ABI=0)


target_link_libraries(
  triton-preprocess-backend
  PRIVATE
    triton-core-serverapi   # from repo-core
    triton-core-serverstub  # from repo-core
    triton-backend-utils    # from repo-backend
)